#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using System;
using System.Collections.Generic;
using NoeticTools.StateEngine.Rules;


namespace NoeticTools.StateEngine.States
{
    public class StateMap<TContext, TState, TStateEvent> :
        IStateMap<TContext, TState, TStateEvent>
    {
        private readonly IStateRulesFactory<TContext, TState, TStateEvent> stateRulesFactory;

        private readonly Dictionary<TState, IStateRules<TContext, TStateEvent>> states =
            new Dictionary<TState, IStateRules<TContext, TStateEvent>>();

        public StateMap(IStateRulesFactory<TContext, TState, TStateEvent> stateRulesFactory)
        {
            this.stateRulesFactory = stateRulesFactory;
        }

        public IStateRules<TContext, TStateEvent> GetStateRules(TState state)
        {
            IStateRules<TContext, TStateEvent> stateRules;

            if (!states.ContainsKey(state))
            {
                stateRules = stateRulesFactory.Create(state);
                states.Add(state, stateRules);
            }
            else
            {
                stateRules = states[state];
            }

            return stateRules;
        }

        public IStateRules<TContext, TStateEvent> GetAllStateRules()
        {
            var rules = new List<IStateRules<TContext, TStateEvent>>();

            var allStates = Enum.GetValues(typeof (TState));

            foreach (TState state in allStates)
            {
                rules.Add(GetStateRules(state));
            }

            return new AllStatesRules<TContext, TStateEvent>(rules.ToArray());
        }
    }
}